package JUC.阻塞队列;

import java.util.concurrent.BlockingQueue;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * @author CKC
 * @DateTime 2021/4/1 20:35
 * <p>
 * volatile/CAS/AtomicInteger/BlockQueue/线程交互/原子引用
 */
public class ProdConsumer_BlockQueueDemo {
    public static void main(String[] args) {
        MyResource myResource = new MyResource(new LinkedBlockingQueue<>(10));
        
        new Thread(() -> {
            System.out.println(Thread.currentThread().getName() + "\t 生产线程启动");
            
            try {
                myResource.myProd();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }, "Prod ").start();
        
        new Thread(() -> {
            System.out.println(Thread.currentThread().getName() + "\t 消费线程启动\n");
            
            try {
                myResource.myConsumer();
                System.out.println("\n");
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }, "Consumer ").start();
        
        
        try { TimeUnit.SECONDS.sleep(5); } catch (InterruptedException e) { e.printStackTrace(); }
        System.out.println("\n\n5秒钟时间到，大老板 main 线程叫停，活动结束");
        myResource.stop();
    }
}


class MyResource {
    private volatile boolean FLAG = true;//默认开启，进行生产+消费
    private AtomicInteger atomicInteger = new AtomicInteger();
    
    BlockingQueue<String> blockingQueue = null;
    
    public MyResource(BlockingQueue<String> blockingQueue) {
        this.blockingQueue = blockingQueue;
        System.out.println(blockingQueue.getClass().getName());//查看实现BlockingQueue的实现类的名称
    }
    
    public void myProd() throws InterruptedException {
        String data = null;
        Boolean retValue = false;
        while (FLAG) {
            data = atomicInteger.incrementAndGet() + "";
            retValue = blockingQueue.offer(data, 2L, TimeUnit.SECONDS);
            if (retValue) {
                System.out.println(Thread.currentThread().getName() + "\t 插入队列  " + data + "  成功");
            } else {
                System.out.println(Thread.currentThread().getName() + "\t 插入队列  " + data + "  失败");
            }
            TimeUnit.SECONDS.sleep(1);
        }
        System.out.println(Thread.currentThread().getName() + "\t 大老板叫停，表示FLAG=false，生产动作结束");
    }
    
    
    public void myConsumer() throws InterruptedException {
        String result = null;
        while (FLAG) {
            result = blockingQueue.poll(2L, TimeUnit.SECONDS);
            if (null == result || result.equalsIgnoreCase("")) {
                FLAG = false;
                System.out.println(Thread.currentThread().getName() + "\t 超时2秒钟没有取到值，消费退出");
                System.out.println();
                System.out.println();
                return;
            }
            System.out.println(Thread.currentThread().getName() + "\t 消费队列：" + result + "  成功");
        }
    }
    
    public void stop() {
        this.FLAG = false;
    }
}